import json

import pandas as pd

from req import *

# 假设data是直接从某处（如文件、网络请求等）获取的，但在这里我们直接定义它
# 注意：这里data是一个列表，它包含了一个列表（即你的数据）
# data1 = [
#     [
#         "123",
#         {
#             "question_text": "aaaaa",
#             "flag": "0"
#         },
#         {
#             "question_text": "bbbbb",
#             "flag": "1"
#         },
#         {
#             "question_text": "ccccc",  # 我修改了这里以避免重复，但这不是必需的
#             "flag": "1"
#         }
#     ],
#     [
#         'abc',
#         {
#             "question_text": "aaaaa",
#             "flag": "2"
#         },
#         {
#             "question_text": "bbbbb",
#             "flag": "2"
#         },
#         {
#             "question_text": "ccccc",  # 我修改了这里以避免重复，但这不是必需的
#             "flag": "1"
#         }
#     ]
# ]

def Output_Excel(data):
    question_name = data[0]
    # print(question_name)
    columns = ['用户名称']
    for num in range(1,len(data[0])):
        columns.append(list(data[0][num].values())[0])
    # print(columns)
    sum_list=[]
    for i in range(len(data)):
        a_list=[]
        user_id=data[i].pop(0)
        a_list.append(user_id)
        for j in range(len(data[i])):
            question=data[i][j]
            a_list.append(question['flag'])
        sum_list.append(a_list)
    # print(sum_list)
    df=pd.DataFrame(sum_list, columns=columns)
    df.to_excel('output.xlsx', index=False)


# 数据处理
def json_to_entity(jsonData,jsonUserId):
    data=[json.loads(v) for v in jsonData]
    for i in range(len(data)):
        data[i].insert(0,jsonUserId[i])
    return data


# 导出为exc
def output_excel(str):
    data1, data2 = sum_data_qt_req(str)
    data = json_to_entity(data1, data2)
    Output_Excel(data)
